Wei+, 2022, TMLR